# -*- coding: utf-8 -*-
# !/usr/bin/env python
"""
-------------------------------------------------
   File Name：     config
   Description :   
   Author :       lth
   date：          2022/6/24
-------------------------------------------------
   Change Activity:
                   2022/6/24 18:17: create this script
-------------------------------------------------
"""
__author__ = 'lth'


class GetConfig:
    project_name = "RetinaFace Detection"
    use_cuda = True
    seed = 123
    resume = True
    pretrained_weight = "./weights/best.pth"

    lr = 1e-3
    milestone = [5, 10, 15, 20, 25, 30, 35, 40, 45, 50, 55, 60, 65, 70, 75, 80]
    epochs = 200

    train_batch_size = 1
    test_batch_size = 1
    target_dir = "face_mask"
    annotations = "./data/" + target_dir + "/annotations.txt"

